﻿using System;
using System.Collections.Generic;
using System.Data;
using System.IO;
using System.Text;
using System.Web;
using NPOI;
using NPOI.HPSF;
using NPOI.HSSF;
using NPOI.HSSF.UserModel;
using NPOI.HSSF.Util;
using NPOI.POIFS;
using NPOI.Util;
using NPOI.SS.UserModel;
using NPOI.SS.Util;

namespace MyCommon
{
    /// <summary>
    /// OFFICE处理文件 导出，导入
    /// </summary>
    public class MyOffice
    {
        #region DataTable导出到Excel文件
        /// <summary>  
        /// DataTable导出到Excel文件  
        /// </summary>  
        /// <param name="dtSource">源DataTable</param>  
        /// <param name="strHeaderText">表头文本</param>  
        /// <param name="strFileName">保存位置</param>  
        public static void Export(DataTable dtSource, string strHeaderText, string strFileName)
        {
            using (MemoryStream ms = Export(dtSource, strHeaderText, new MyFileInfo()))
            {
                using (FileStream fs = new FileStream(strFileName, FileMode.Create, FileAccess.Write))
                {
                    byte[] data = ms.ToArray();
                    fs.Write(data, 0, data.Length);
                    fs.Flush();
                }
            }
        }
        #endregion

        #region DataTable导出到Excel文件，可以填写文件信息
        /// <summary>  
        /// DataTable导出到Excel文件  
        /// </summary>  
        /// <param name="dtSource">源DataTable</param>  
        /// <param name="strHeaderText">表头文本</param>  
        /// <param name="strFileName">保存位置</param>  
        /// <param name="info">FileInfo 实体 文件生成时的文件的属性信息</param>
        public static void Export(DataTable dtSource, string strHeaderText, string strFileName, MyFileInfo info)
        {
            using (MemoryStream ms = Export(dtSource, strHeaderText, info))
            {
                using (FileStream fs = new FileStream(strFileName, FileMode.Create, FileAccess.Write))
                {
                    byte[] data = ms.ToArray();
                    fs.Write(data, 0, data.Length);
                    fs.Flush();
                }
            }
        }
        #endregion

        #region 用于Web导出execl
        /// <summary>  
        /// 用于Web导出execl 
        /// </summary>  
        /// <param name="dtSource">源DataTable，列要处理一下</param>  
        /// <param name="strHeaderText">表头文本</param>  
        /// <param name="strFileName">文件名，必须加后缀 .xls</param>  
        public static void ExportByWeb(DataTable dtSource, string strHeaderText, string strFileName)
        {

            HttpContext curContext = HttpContext.Current;

            // 设置编码和附件格式  
            curContext.Response.ContentType = "application/vnd.ms-excel";
            curContext.Response.ContentEncoding = Encoding.UTF8;
            curContext.Response.Charset = "";
            if (MyUrl.GetBrowser().IndexOf("ie") != -1)
            {
                curContext.Response.AppendHeader("Content-Disposition",
                "attachment;filename=" + HttpUtility.UrlEncode(strFileName, Encoding.UTF8));
            }
            else
            {
                curContext.Response.AppendHeader("Content-Disposition",
                    "attachment;filename=" + strFileName);
            }

            curContext.Response.BinaryWrite(Export(dtSource, strHeaderText, new MyFileInfo()).GetBuffer());
            curContext.Response.End();

        }
        #endregion

        #region 用于Web导出execl
        /// <summary>  
        /// 用于Web导出execl 
        /// </summary>  
        /// <param name="dtSource">源DataTable</param>  
        /// <param name="strHeaderText">表头文本</param>  
        /// <param name="strFileName">文件名</param>  
        /// <Author>柳永法 http://www.yongfa365.com/ 2010-5-8 22:21:41</Author>  
        public static void ExportByWeb(DataTable dtSource, string strHeaderText, string strFileName, MyFileInfo info)
        {

            HttpContext curContext = HttpContext.Current;

            // 设置编码和附件格式  
            curContext.Response.ContentType = "application/vnd.ms-excel";
            curContext.Response.ContentEncoding = Encoding.UTF8;
            curContext.Response.Charset = "";
            //curContext.Response.AppendHeader("Content-Disposition",
            //    "attachment;filename=" + HttpUtility.UrlEncode(strFileName, Encoding.UTF8));
            curContext.Response.AppendHeader("Content-Disposition",
                "attachment;filename=" + strFileName);

            curContext.Response.BinaryWrite(Export(dtSource, strHeaderText, info).GetBuffer());
            curContext.Response.End();

        }
        #endregion

        #region DataTable导出到Excel的MemoryStream
        /// <summary>  
        /// DataTable导出到Excel的MemoryStream  
        /// </summary>  
        /// <param name="dtSource">源DataTable</param>  
        /// <param name="strHeaderText">表头文本</param>  
        /// <param name="info">FileInfo 实体 文件生成时的文件的属性信息</param>
        public static MemoryStream Export(DataTable dtSource, string strHeaderText, MyFileInfo info)
        {
            HSSFWorkbook workbook = new HSSFWorkbook();
            var sheet = workbook.CreateSheet();

            #region 右击文件 属性信息
            {
                DocumentSummaryInformation dsi = PropertySetFactory.CreateDocumentSummaryInformation();
                dsi.Company = "http://www.yongfa365.com/";
                workbook.DocumentSummaryInformation = dsi;

                SummaryInformation si = PropertySetFactory.CreateSummaryInformation();
                si.Author = info.Author; //填加xls文件作者信息  
                si.ApplicationName = info.ApplicationName; //填加xls文件创建程序信息  
                si.LastAuthor = info.LastAuthor; //填加xls文件最后保存者信息  
                si.Comments = info.Comments; //填加xls文件作者信息  
                si.Title = info.Title; //填加xls文件标题信息  
                si.Subject = info.Subject;//填加文件主题信息  
                si.CreateDateTime = info.CreateDateTime;
                workbook.SummaryInformation = si;
            }
            #endregion

            var dateStyle = workbook.CreateCellStyle();
            var format = workbook.CreateDataFormat();
            dateStyle.DataFormat = format.GetFormat("yyyy-mm-dd");

            //取得列宽  
            int[] arrColWidth = new int[dtSource.Columns.Count];
            foreach (DataColumn item in dtSource.Columns)
            {
                arrColWidth[item.Ordinal] = Encoding.GetEncoding(936).GetBytes(item.ColumnName.ToString()).Length;
            }
            for (int i = 0; i < dtSource.Rows.Count; i++)
            {
                for (int j = 0; j < dtSource.Columns.Count; j++)
                {
                    int intTemp = Encoding.GetEncoding(936).GetBytes(dtSource.Rows[i][j].ToString()).Length;
                    if (intTemp > arrColWidth[j])
                    {
                        arrColWidth[j] = intTemp;
                    }
                }
            }



            int rowIndex = 0;

            foreach (DataRow row in dtSource.Rows)
            {
                #region 新建表，填充表头，填充列头，样式
                if (rowIndex == 65535 || rowIndex == 0)
                {
                    if (rowIndex != 0)
                    {
                        sheet = workbook.CreateSheet();
                    }

                    #region 表头及样式
                    {
                        var headerRow = sheet.CreateRow(0);
                        headerRow.HeightInPoints = 25;
                        headerRow.CreateCell(0).SetCellValue(strHeaderText);

                        var headStyle = workbook.CreateCellStyle();
                        headStyle.Alignment = HorizontalAlignment.CENTER;
                        var font = workbook.CreateFont();
                        font.FontHeightInPoints = 20;
                        font.Boldweight = 700;
                        headStyle.SetFont(font);

                        headerRow.GetCell(0).CellStyle = headStyle;

                        sheet.AddMergedRegion(new CellRangeAddress(0, 0, 0, dtSource.Columns.Count - 1));

                    }
                    #endregion


                    #region 列头及样式
                    {
                        var headerRow = sheet.CreateRow(1);


                        var headStyle = workbook.CreateCellStyle();
                        headStyle.Alignment = HorizontalAlignment.CENTER;
                        var font = workbook.CreateFont();
                        font.FontHeightInPoints = 10;
                        font.Boldweight = 700;
                        headStyle.SetFont(font);


                        foreach (DataColumn column in dtSource.Columns)
                        {
                            headerRow.CreateCell(column.Ordinal).SetCellValue(column.ColumnName);
                            headerRow.GetCell(column.Ordinal).CellStyle = headStyle;

                            //设置列宽  
                            sheet.SetColumnWidth(column.Ordinal, (arrColWidth[column.Ordinal] + 1) * 256);

                        }
                        //headerRow.Dispose();
                    }
                    #endregion

                    rowIndex = 2;
                }
                #endregion


                #region 填充内容
                var dataRow = sheet.CreateRow(rowIndex);
                foreach (DataColumn column in dtSource.Columns)
                {
                    var newCell = dataRow.CreateCell(column.Ordinal);

                    string drValue = row[column].ToString();

                    switch (column.DataType.ToString())
                    {
                        case "System.String"://字符串类型  
                            newCell.SetCellValue(drValue);
                            break;
                        case "System.DateTime"://日期类型  
                        case "MySql.Data.Types.MySqlDateTime": //MySql类型
                            if (drValue == "0000/0/0 0:00:00" || String.IsNullOrEmpty(drValue))
                            {
                                //当时间为空，防止生成的execl 中是一串“#######”号，所有赋值为空字符串
                                newCell.SetCellValue("");
                            }
                            else
                            {
                                DateTime dateV;
                                DateTime.TryParse(drValue, out dateV);
                                newCell.SetCellValue(dateV);

                                newCell.CellStyle = dateStyle;//格式化显示  
                            }
                            break;
                        case "System.Boolean"://布尔型  
                            bool boolV = false;
                            bool.TryParse(drValue, out boolV);
                            newCell.SetCellValue(boolV);
                            break;
                        case "System.Int16"://整型  
                        case "System.Int32":
                        case "System.Int64":
                        case "System.Byte":
                            int intV = 0;
                            int.TryParse(drValue, out intV);
                            newCell.SetCellValue(intV);
                            break;
                        case "System.Decimal"://浮点型  
                        case "System.Double":
                            double doubV = 0;
                            double.TryParse(drValue, out doubV);
                            newCell.SetCellValue(doubV);
                            break;
                        case "System.DBNull"://空值处理  
                            newCell.SetCellValue("");
                            break;
                        default:
                            newCell.SetCellValue("");
                            break;
                    }

                }
                #endregion

                rowIndex++;
            }

            #region 添加图片
            //byte[] bytes = System.IO.File.ReadAllBytes(@"E:\work\手机维修门店服务端项目管理\12_开发代码\PhoneStation\PSWeb\UploadFile\App\Images\web\020120509010041613.jpg");
            //int pictureIdx = workbook.AddPicture(bytes, PictureType.JPEG);

            //// Create the drawing patriarch.  This is the top level container for all shapes. 
            //IDrawing patriarch = sheet.CreateDrawingPatriarch();

            ////add a picture
            //HSSFClientAnchor anchor = new HSSFClientAnchor(0, 0, 1023, 0, 1, rowIndex+1, 11, 13);
            //IPicture pict = patriarch.CreatePicture(anchor, pictureIdx);
            //pict.Resize();
            #endregion

            using (MemoryStream ms = new MemoryStream())
            {
                workbook.Write(ms);
                ms.Flush();
                ms.Position = 0;

                //sheet.Dispose();
                //workbook.Dispose();//一般只用写这一个就OK了，他会遍历并释放所有资源，但当前版本有问题所以只释放sheet  
                return ms;
            }

        }

        #endregion

        #region 读取excel 默认第一行为标头
        /// <summary>读取excel  
        /// 读取excel 默认第一行为标头  
        /// </summary>  
        /// <param name="strFileName">excel文档路径</param>  
        /// <returns></returns>  
        public static DataTable Import(string strFileName)
        {
            DataTable dt = new DataTable();

            HSSFWorkbook hssfworkbook;
            using (FileStream file = new FileStream(strFileName, FileMode.Open, FileAccess.Read))
            {
                hssfworkbook = new HSSFWorkbook(file);
            }
            ISheet sheet = hssfworkbook.GetSheetAt(0);
            System.Collections.IEnumerator rows = sheet.GetRowEnumerator();

            IRow headerRow = sheet.GetRow(0);
            int cellCount = headerRow.LastCellNum;

            for (int j = 0; j < cellCount; j++)
            {
                ICell cell = headerRow.GetCell(j);
                dt.Columns.Add(cell.ToString());
            }

            for (int i = (sheet.FirstRowNum + 1); i <= sheet.LastRowNum; i++)
            {
                IRow row = sheet.GetRow(i);
                DataRow dataRow = dt.NewRow();

                for (int j = row.FirstCellNum; j < cellCount; j++)
                {
                    if (row.GetCell(j) != null)
                        dataRow[j] = row.GetCell(j).ToString();
                }

                dt.Rows.Add(dataRow);
            }
            return dt;
        }
        #endregion

    }

    #region 导出文件时 文件的属性信息 model
    /// <summary>
    /// 导出文件时 文件的属性信息
    /// </summary>
    public class MyFileInfo
    {
        /// <summary>
        /// 填加xls文件作者信息 
        /// </summary>
        public String Author;

        /// <summary>
        /// 填加xls文件创建程序信息
        /// </summary>
        public String ApplicationName;

        /// <summary>
        /// 填加xls文件最后保存者信息
        /// </summary>
        public String LastAuthor;

        /// <summary>
        /// 填加xls文件作者信息
        /// </summary>
        public String Comments;

        /// <summary>
        /// 填加xls文件标题信息  
        /// </summary>
        public String Title;

        /// <summary>
        /// 填加文件主题信息
        /// </summary>
        public String Subject;

        /// <summary>
        /// 创建时间
        /// </summary>
        public DateTime CreateDateTime;

        /// <summary>
        /// 构造函数
        /// </summary>
        public MyFileInfo()
        {
            Author = "未知"; //填加xls文件作者信息  
            ApplicationName = "未知"; //填加xls文件创建程序信息 
            LastAuthor = "未知"; //填加xls文件最后保存者信息  
            Comments = "未知"; //填加xls文件作者信息  
            Title = "未知"; //填加xls文件标题信息  
            Subject = "未知";//填加文件主题信息  
            CreateDateTime = DateTime.Now;
        }

    }
    #endregion
}
